#!/bin/bash -l
#SBATCH --nodes=1
#SBATCH --gres=gpu:1
#SBATCH -p g-v100-1
#SBATCH --cpus-per-task=10
#SBATCH --ntasks-per-node=1

#activate conda env
source activate dp

export path=/home/cloudam/aaa/src/Proteus-Legacy/data/common_proteins

THEANO_FLAGS='floatX=float32,device=cuda' python -u /home/cloudam/aaa/src/predict.py\
    --dataset-input $path